# modellaboratory

from langchain.chains.llm import LLMChain
# from langchain.llms import openai, cohere, huggingface_hub
from langchain_community.llms import openai, cohere, huggingface_hub
from langchain.prompts import PromptTemplate
from langchain.model_laboratory import ModelLaboratory
from langchain_ollama import ChatOllama

# 配置模型信息
llms = [
    # openai(temperature = 0),
    # cohere(model = "command-xlarge-20221108", max_tokens = 20,
    #        temperature = 0),
    # huggingface_hub(repo_id = "google/flan-t5-x1",
    #                 model_kwargs = {"temperature": 1})
    ChatOllama(model = "deepseek-r1:7b"),
    ChatOllama(model = "qwen:1.8b")
]

model_lab = ModelLaboratory.from_llms(llms)
model_lab.compare("what color is a flamingo?")